# config.py

# --- Model Configuration ---
BASE_EMBED_MODEL = "local:BAAI/bge-small-en-v1.5"

# --- Dataset Configuration ---
INITIAL_DATASET_NAME = "scifact"
INSERTION_DATASET_NAME = "nfcorpus"
INSERTION_ID_PREFIX = "nfcorpus"  # 防止ID冲突的前缀

# --- Finetuning & Evaluation Parameters ---
RETRIEVER_TOP_K = 10
ADAPTER_FINETUNE_EPOCHS = 4
ADAPTER_FINETUNE_BATCH_SIZE = 10
USE_NUDGE_N = True  # NUDGE specific parameter
DEVICE = "cuda"  # "cuda" or "cpu"